import os
class Files:
    def createDir(self,path):
        if not os.path.isdir(path):
            os.makedirs(path)

def list_dir(file_dir):
    '''
        通过 listdir 得到的是仅当前路径下的目录名，不包括子目录中的目录，如果需要得到所有目录需要递归
    '''
    dir_list = os.listdir(file_dir)
    tmplist=[]
    for cur_file in dir_list:
        # 获取文件的绝对路径
        path = os.path.join(file_dir, cur_file)
        #if os.path.isfile(path): # 判断是否是文件还是目录需要用绝对路径
        #   print("{0} : is file!".format(cur_file))
        if os.path.isdir(path):
            #print("{0} : is dir!".format(cur_file))
            tmplist.append(int(cur_file))
            #list_dir(path) # 递归子目录
    return tmplist

def list_file(file_dir,all00list,allxxlist):
    '''
        通过 listdir 得到的是仅当前路径下的文件名(末尾为_0_0的放在all00list，否则放在allxxlist里)，包括子目录中的文件
    '''
    dir_list = os.listdir(file_dir)
    for cur_file in dir_list:
        # 获取文件的绝对路径
        path = os.path.join(file_dir, cur_file)
        if os.path.isfile(path): # 判断是否是文件还是目录需要用绝对路径
            #print("{0} : is file!".format(cur_file))
            tmpfilename=cur_file.replace('.csv','')
            if tmpfilename[-4:]=="_0_0": #截取倒数第4位到结尾
                all00list.append(tmpfilename)
            else:
                allxxlist.append(tmpfilename)
        if os.path.isdir(path):
            list_file(path,all00list,allxxlist) # 递归子目录
    return all00list,allxxlist